/*
    该文件负责实现compiler类，compiler类功能是对源文件执行编译
*/

#pragma once
#include <iostream>
#include <string>
#include <sys/types.h>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <fcntl.h>
#include "../comm/util.hpp"
#include "../comm/log.hpp"

namespace ns_compiler
{
    using namespace ns_log;

    class Compiler
    {
    public:
        Compiler(){}
        ~Compiler(){}

        // 接收一个参数：源文件名，然后对源文件进行编译，生成可执行文件或者编译失败生成错误文件
        static bool Compile(const std::string &file_name)
        {
            pid_t fd = fork();
            if(fd < 0) // 创建子进程失败
            {
                LOG(ERROR) << "创建子进程失败" << '\n';
                exit(1);
            }
            else if(fd == 0)
            {
                // 子进程

                // 错误输出重定向
                umask(0);
                int stderr_id = open(ns_util::PathUtil::CompileErr(file_name).c_str(), O_CREAT | O_WRONLY, 0644);
                if(stderr_id < 0) // 打开失败
                {
                    LOG(ERROR) << "编译错误文件打开失败" << '\n';
                    exit(2);
                }
                dup2(stderr_id, 2); // 将标准错误输错重定向到临时错误文件

                // 执行进程替换，使用g++对源码文件进行编译
                // g++ -o target src -std=c++11
                execlp("g++", "g++", "-o", ns_util::PathUtil::Exe(file_name).c_str(), 
                    ns_util::PathUtil::Src(file_name).c_str(), "-D", "COMPILE_ONLINE", "-std=c++11", nullptr);
                
                LOG(ERROR) << "编译器编译失败" << '\n';
                exit(3);
            }
            else
            {
                // 父进程

                // 阻塞等待子进程
                waitpid(fd, nullptr, 0);

                if(ns_util::FileUtil::IsExists(ns_util::PathUtil::Exe(file_name)))
                {
                    // 可执行文件存在，编译成功
                    LOG(INFO) << " -> " << ns_util::PathUtil::AddSuffix(file_name, ".cc") << " 编译成功!" << '\n';
                    return true;
                }
                
                // 可执行文件不存在，编译失败
                LOG(ERROR) << " -> 编译失败，可执行文件不存在!" << '\n';
                return false;
            }
        }
    };
}